python train.py --batch_size 200 \
                --gpu 1,3 \
                --num_epochs 30 \
                --lr 0.00001 \
                --dataset Pose_300W_LP \
                --data_dir /home/algtest/datasets/300W/300W_LP \
                --filename_list /home/algtest/datasets/300W/300W_LP/files.txt \
                --dataset_test AFLW2000 \
                --data_dir_test /home/algtest/datasets/AFLW2000 \
                --filename_list_test /home/algtest/datasets/AFLW2000/files.txt \
                --backbone_name RepVGG-B1g2 \
                --backbone_file /home/algtest/pretrained_models/6drep/RepVGG-B1g2-train.pth \
                --steps_eval 500 \
                --log_dir ./log \
                --output_dir ./output \
